/*
 *   Copyright 2021 zzh
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 */

package com.aduib.boot.logging.filter.MDC;

import cn.hutool.core.util.IdUtil;
import com.aduib.boot.common.util.WebUtils;
import com.aduib.boot.logging.constants.LogConstants;
import com.aduib.boot.logging.filter.AbstractLogFilter;
import org.slf4j.MDC;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

/**
 * @description: 利用 Slf4J 的 MDC 功能，记录 TraceId
 * @author: zzh
 * @date: 2021/9/17 09:55
 */
public class TraceIdFilter extends AbstractLogFilter {

  @Override
  public boolean beforeRequest(HttpServletRequest request, HttpServletResponse response,
      FilterChain filterChain) throws ServletException, IOException {
    String objectId = IdUtil.objectId();
    MDC.put(LogConstants.TRACE_ID, objectId);
    WebUtils.addAttribute(request,LogConstants.TRACE_ID, objectId);
    try {
      filterChain.doFilter(request, response);
    }
    finally {
      MDC.remove(LogConstants.TRACE_ID);
      WebUtils.removeAttribute(request,LogConstants.TRACE_ID);
    }
    return false;
  }

  @Override
  public void afterRequest(HttpServletRequest request, HttpServletResponse response,
      Throwable throwable) throws ServletException, IOException {

  }
}
